// #include <iostream>    // 包含输入输出流头文件
// using namespace std;    // 使用标准命名空间

#include <workManager.h>

WorkManager::WorkManager()
{
    // 1、文件不存在
    ifstream ifs;
    ifs.open(FILENAME, ios::in);    // 读文件

    if (!ifs.is_open())
    {
        cout << "文件不存在" << endl;

        // 初始化属性
        // 初始化记录人数
        this->m_EmpNum = 0;

        // 初始化数组指针
        this->m_EmpArray = NULL;

        // 初始化文件是否为空
        this->m_FileIsEmpty = true;
        ifs.close();
        return;
    }

    // 2、文件存在 数据为空
    char ch;
    ifs >> ch;
    if (ifs.eof())
    {
        // 文件为空
        cout << "文件为空" << endl;
         
        // 初始化记录人数
        this->m_EmpNum = 0;

        // 初始化数组指针
        this->m_EmpArray = NULL;

        // 初始化文件是否为空
        this->m_FileIsEmpty = true; 
        ifs.close();
        return;
    }

    // 3、文件存在，并且记录数据
    int num = this->get_EmpNum();
    // cout << "职工人数为：" << num << endl;
    this->m_EmpNum = num;

    // 开辟空间
    this->m_EmpArray = new Worker*[this->m_EmpNum];

    // 将文件中的数据，存到数组中
    this->init_Emp();

    // for (int i = 0; i < this->m_EmpNum; i++)
    // {
    //     cout << "职工编号：" << this->m_EmpArray[i]->m_Id
    //         << "\t职工姓名：" << this->m_EmpArray[i]->m_Name
    //         << "\t部门编号：" << this->m_EmpArray[i]->m_DeptId << endl;
    // }
    
}

// 展示菜单
void WorkManager::Show_Menu()
{
    cout << "欢迎使用职工管理系统：" << endl;
    cout << "0.退出管理程序" << endl;
    cout << "1.增加职工信息" << endl;
    cout << "2.显示职工信息" << endl;
    cout << "3.删除离职职工" << endl;
    cout << "4.修改职工信息" << endl;
    cout << "5.查找职工信息" << endl;
    cout << "6.按照编号排序" << endl;
    cout << "7.清空所有文档" << endl;
    cout << endl;
}

void WorkManager::ExitSystem()
{
    cout << "欢迎下次使用" << endl;
    system("pause");
    exit(0);
}

void WorkManager::Add_Emp()
{
    cout << "请输入添加的职工的数量：" << endl;

    int addNum = 0; // 保存用户的输入数量
    cin >> addNum;

    if (addNum > 0)
    {
        // 添加
        // 计算添加后的新空间大小
        int newSize = this->m_EmpNum + addNum;  // 新空间的大小 = 原来人数+新增人数

        // 开辟新空间
        Worker ** newSpace = new Worker*[newSize];

        // 将原来的空间中的数据，拷贝到新空间中
        if (this->m_EmpArray != NULL)
        {
            for(int i = 0; i < this->m_EmpNum; i++)
            {
                newSpace[i] = this->m_EmpArray[i];
            }
        }

        // 添加新数据
        for (int i = 0; i < addNum; i++)
        {
            int id; // 职工编号
            string name;
            int dSelect;

            cout << "请输入第" << i+1 << "个新入职员工的编号：" << endl;
            cin >> id;

            cout << "请输入第" << i+1 << "个新入职员工的姓名：" << endl;
            cin >> name;

            cout << "请选择该职工的岗位：" << endl;
            cout << "1、普通职工" << endl;
            cout << "2、经理" << endl;
            cout << "3、老板" << endl;
            cin >> dSelect;

            Worker * worker = NULL;
            switch (dSelect)
            {
            case 1:
                worker = new Employee(id, name, dSelect);
                break;

            case 2:
                worker = new Manager(id, name, dSelect);
                break;

            case 3:
                worker = new Boss(id, name, dSelect);
                break;
            
            default:
                break;
            }

            // 将创建职工职责，保存到数组中
            newSpace[this->m_EmpNum + i] = worker;
        }

        // 释放原有的空间
        delete [] this->m_EmpArray;

        // 更改新空间的指向
        this->m_EmpArray = newSpace;

        // 更新一下新的职工人数
        this->m_EmpNum = newSize;

        // 更新职工不为空标志
        this->m_FileIsEmpty = false;

        cout << "成功添加了" << addNum << "名新职工" << endl;

        // 保存数据到文件中  
        this->save();
    }
    else
    {
        cout << "输入数据错误" << endl;
    } 

    system("pause");
    system("cls");
}

// 保存文件
void WorkManager::save()
{
    ofstream ofs;
    ofs.open(FILENAME, ios::out);

    for (int i = 0; i < this->m_EmpNum; i++)
    {
        ofs << this->m_EmpArray[i]->m_Id << " "
            << this->m_EmpArray[i]->m_Name << " "
            << this->m_EmpArray[i]->m_DeptId << " " << endl;
    }

    ofs.close();
}

int WorkManager::get_EmpNum()
{
    ifstream ifs;
    ifs.open(FILENAME, ios::in);    // 打开文件

    int id;
    string name;
    int dId;

    int num = 0;
    while (ifs >> id && ifs >> name && ifs >> dId)
    {
        num++;
    }

    return num;
}

void WorkManager::init_Emp()
{
    ifstream ifs;
    ifs.open(FILENAME, ios::in);

    int id;
    string name;
    int dId;

    int index = 0;
    while (ifs >> id && ifs >> name && ifs >> dId)
    {
        Worker * worker = NULL;

        if (dId == 1)   // 普通职工
        {
            worker = new Employee(id, name, dId);
        }
        else if (dId == 2)
        {
            worker = new Manager(id, name, dId);
        }
        else
        {
            worker = new Boss(id, name, dId);
        }

        this->m_EmpArray[index] = worker;
        index++;
    }

    ifs.close();
}

void WorkManager::Show_Emp()
{
    if (this->m_FileIsEmpty)
    {
        cout << "文件不存在或记录为空！" << endl;
    }
    else
    {
        for (int i = 0; i < m_EmpNum; i++)
        {
            // 利用多态调用程序接口
            this->m_EmpArray[i]->showInfo();
        }
        system("pause");
        system("cls"); 
    }
}

void WorkManager::Del_Emp()
{
    if (this->m_FileIsEmpty)
    {
        cout << "文件不存在或记录为空！" << endl;
    }
    else
    {
        // 按照职工编号删除
        cout << "请输入想要删除职工编号：" << endl;
        int id = 0;
        cin >> id;

        int index = this->IsExist(id);
        
        if (index != -1)
        { 
            // 数据前移 
            for (int i = 0; i < this->m_EmpNum-1; i++)
            {
                this->m_EmpArray[i] = this->m_EmpArray[i+1];
            }
            // 更新数组中记录人员个数
            this->m_EmpNum--;
            this->save();

            cout << "删除成功！" << endl;
        }
        else
        {
            cout << "删除失败，未找到该职工" << endl;
        }
    }

    system("pause");
    system("cls");  
}

int WorkManager::IsExist(int id)
{
    int index = -1;

    for (int i = 0; i < this->m_EmpNum; i++)
    {
        if (this->m_EmpArray[i]->m_Id == id)
        {
            // 找到职工
            index = i;
            break;
        }
    }

    return index;
}

void WorkManager::Mod_Emp()
{
    if (this->m_FileIsEmpty)
    {
        cout << "文件不存在或记录为空！" << endl;
    }
    else
    {
        cout << "请输入修改职工的编号：" << endl;
        int id;
        cin >> id;

        int ret = this->IsExist(id);
        if (ret != -1)
        {
            // 查找到编号的职工
            delete this->m_EmpArray[ret];

            int newId = 0;
            string newName = "";
            int dSelect = 0;

            cout << "查到：" << id << "号职工：请输入新职工号：" << endl;
            cin >> newId;

            cout << "请输入新姓名：" << endl;
            cin >> newName;

            cout << "请输入岗位" << endl;
            cout << "1、普通职工" << endl;
            cout << "2、经理" << endl;
            cout << "3、老板" << endl;

            cin >> dSelect;

            Worker * worker = NULL;
            switch (dSelect)
            {
            case 1:
                worker = new Employee(newId, newName, dSelect);
                break;

            case 2:
                worker = new Manager(newId, newName, dSelect);
                break;

            case 3:
                worker = new Boss(newId, newName, dSelect);
                break;
            
            default:
                break;
            }

            // 更新数据到数组中
            this->m_EmpArray[ret] = worker;

            cout << "修改成功！" << endl;

            // 保存到文件中
            this->save();
        }
        else
        {
            cout << "修改失败，查无此人！" << endl;
        }
    }

    system("pause");
    system("cls");
}

void WorkManager::Find_Emp()
{
    if (this->m_FileIsEmpty)
    {
        cout << "文件不存在或记录为空！" << endl;
        return;
    }

    cout << "请出入查找的方式：" << endl;
    cout << "1、按照职工编号查找：" << endl;
    cout << "2、按照职工姓名查找：" << endl;

    int select = 0;
    cin >> select;

    if (select == 1)
    {
        // 按编号查
        int id;
        cout << "请输入查找的职工编号：" << endl;
        cin >> id;

        int ret = IsExist(id);
        if (ret != -1)
        {
            // 找到职工
            cout << "查找成功！该职工的信息如下：" << endl;
            this->m_EmpArray[ret]->showInfo();
        }
        else
        {
            cout << "查找失败，查无此人" << endl;
        }
    }
    else if (select == 2)
    {
        // 按姓名查
        string name;
        cout << "请输入查找的姓名：" << endl;
        cin >> name;

        // 加入判断是否查到的标志
        bool flag = false;

        for (int i = 0; i < this->m_EmpNum; i++)
        {
            if (this->m_EmpArray[i]->m_Name == name)
            {
                cout << "查找成功，职工编号为："
                    << this->m_EmpArray[i]->m_Id
                    << "号职工信息如下：" << endl;

                flag = true;

                this->m_EmpArray[i]->showInfo();
            }
        }
        if (flag == false)
        {
            cout << "查找失败，查无此人！" << endl;
        }
    }
    else
    {
        cout << "输入选项有误！" << endl;
    }

    system("pause");
    system("cls");
}

void WorkManager::Sort_Emp()
{
    if (this->m_FileIsEmpty)
    {
        cout << "文件不存在或记录为空！" << endl;
        system("pause");
        system("cls");
        return;
    }
    
    cout << "请选择排序方式：" << endl;
    cout << "1、按职工号进行升序" << endl;
    cout << "2、按职工号进行降序" << endl;

    int select = 0;
    cin >> select;
    for (int i = 0; i < m_EmpNum; i++)
    {
        int minOrMax = i;   // 声明最小值 或 最大值下标
        for (int j = i + 1; j < this->m_EmpNum; j++)
        {
            if (select == 1)
            {
                if (this->m_EmpArray[minOrMax]->m_Id > this->m_EmpArray[j]->m_Id)
                {
                    minOrMax = j;
                }
            }
            else    // 降序
            {
                if (this->m_EmpArray[minOrMax]->m_Id < this->m_EmpArray[j]->m_Id)
                {
                    minOrMax = j;
                }
            }
        }

        if (i != minOrMax)
        {
            Worker * temp = this->m_EmpArray[i];
            this->m_EmpArray[i] = this->m_EmpArray[minOrMax];
            this->m_EmpArray[minOrMax] = temp;
        }

        cout << "排序成功！排序后的结果为：" << endl;
        this->save();
        this->Show_Emp();
    }
}

void WorkManager::Clean_File()
{
    cout << "确定清空？" << endl;
    cout << "1、确定" << endl;
    cout << "2、返回" << endl;

    int select = 0;
    cin >> select;

    if (select == 1)
    {
        // 清空文件
        ofstream ofs(FILENAME, ios::trunc);
        ofs.close();

        if (this->m_EmpArray != NULL)
        {
            for (int i = 0; i < this->m_EmpNum; i++)
            {
                delete this->m_EmpArray[i];
                this->m_EmpArray[i] = NULL;
            }
            
            // 删除堆区数组指针
            delete[] this->m_EmpArray;
            this->m_EmpArray = NULL;
            this->m_EmpNum = 0;
            this->m_FileIsEmpty = true;
        }

        cout << "清空成功！" << endl;
    }

    system("pause");
    system("cls");
}

WorkManager::~WorkManager()
{
    if (this->m_EmpArray != NULL)
    {
        delete[] this->m_EmpArray;
        this->m_EmpArray = NULL;
    }
    cout << "WorkManager的析构函数" << endl;
}
